package com.xy.core.utils;

import freemarker.template.Configuration;
import freemarker.template.Template;
import freemarker.template.TemplateException;

import java.io.*;
import java.util.Locale;
import java.util.Map;

/**
 * Freemarker 模版引擎类
 * 创建人：FH Q313596790
 * 创建时间：2015年2月8日
 * @version
 */
public class Freemarker {


	/**
	 * 打印到控制台(测试用)
	 *  @param ftlName
	 */
	public static void print(String ftlName, Map<String,Object> root, String ftlPath) throws Exception{
		try {
			Template temp = getTemplate(ftlName, ftlPath);		//通过Template可以将模板文件输出到相应的流
			temp.process(root, new PrintWriter(System.out));
		} catch (TemplateException e) {
			e.printStackTrace();
		} catch (IOException e) {
			e.printStackTrace();
		}
	}
	/**
	 *
	 * @param ftlFileName ftl文件名 ftl文件目录  默认classpath/ftl/目录开始
	 * @param root		传入数据
	 * @param outFile	输出文件名	输出路径，默认是项目目录开始  filepath+outFile 就是文件路径
	 * @throws Exception
	 */
	public static void printFile(String ftlFileName, Map<String,Object> root, String outFile) throws Exception{
		try {
			File file = new File(PathUtil.getWebpath() + outFile);
			if(!file.getParentFile().exists()){				//判断有没有父路径，就是判断文件整个路径是否存在
				file.getParentFile().mkdirs();				//不存在就全部创建
			}
			Writer out = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(file), "utf-8"));
			String ftlPath =  ftlFileName.substring(0,ftlFileName.lastIndexOf("/"));
			String ftlName =  ftlFileName.substring(ftlFileName.lastIndexOf("/")+1);
			Template template = getTemplate2(ftlName, ftlPath);
			template.process(root, out);					//模版输出
			out.flush();
			out.close();
		} catch (TemplateException e) {
			e.printStackTrace();
		} catch (IOException e) {
			e.printStackTrace();
		}
	}

	/**
	 * 通过文件名加载模版
	 * @param ftlName 类名 去掉.class 加上.ftl
	 * @param ftlPath 包名，替换点(.)为斜杠(/)
	 */
	public static Template getTemplate2(String ftlName, String ftlPath) throws Exception{
		Template temp;
		try {

			Configuration cfg = new Configuration();  												//通过Freemaker的Configuration读取相应的ftl
			cfg.setEncoding(Locale.CHINA, "utf-8");
			File file = new File(PathUtil.getWebpath()+ftlPath+"/"+ftlName);
			if(file.exists()){
				cfg.setDirectoryForTemplateLoading(new File(PathUtil.getWebpath()+ftlPath));		//设定去哪里读取相应的ftl模板文件
				temp = cfg.getTemplate(ftlName);												//在模板文件目录中找到名称为name的文件
				return temp;
			}else{
				System.err.println(PathUtil.getWebpath()+ftlPath+"    找不到相关文件");
				return null;
			}

		} catch (IOException e) {
			e.printStackTrace();

		}
		return null;
	}

	/**
	 * 输出到输出到文件
	 * @param ftlName   ftl文件名
	 * @param root		传入的map
	 * @param outFile	输出后的文件全部路径
	 * @param filePath	输出前的文件上部路径
	 */
	public static void printFile(String ftlName, Map<String,Object> root, String outFile, String filePath, String ftlPath) throws Exception{
		try {
			File file = new File(PathUtil.getClasspath() + filePath + outFile);
			if(!file.getParentFile().exists()){				//判断有没有父路径，就是判断文件整个路径是否存在
				file.getParentFile().mkdirs();				//不存在就全部创建
			}
			Writer out = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(file), "utf-8"));
			Template template = getTemplate(ftlName, ftlPath);
			template.process(root, out);					//模版输出
			out.flush();
			out.close();
		} catch (TemplateException e) {
			e.printStackTrace();
		} catch (IOException e) {
			e.printStackTrace();
		}
	}
	
	/**
	 * 通过文件名加载模版
	 * @param ftlName 类名 去掉.class 加上.ftl
	 * @param ftlPath 包名，替换点(.)为斜杠(/)
	 */
	public static Template getTemplate(String ftlName, String ftlPath) throws Exception{
		try {
			Configuration cfg = new Configuration();  												//通过Freemaker的Configuration读取相应的ftl
			cfg.setEncoding(Locale.CHINA, "utf-8");
			cfg.setDirectoryForTemplateLoading(new File(PathUtil.getClassResources()+"/ftl/"+ftlPath));		//设定去哪里读取相应的ftl模板文件
			Template temp = cfg.getTemplate(ftlName);												//在模板文件目录中找到名称为name的文件
			return temp;
		} catch (IOException e) {
			e.printStackTrace();
		}
		return null;
	}
}
